package com.boge.ai.rag;

import com.boge.ai.utils.LLMUtils;

import java.io.IOException;

public class Demo02 {
    public static void main(String[] args) throws IOException {
        // 通过外部的知识库，检索到了今天天气相关的信息
        // 我把最近一年的全国主要城市的天气信息都查询出来  北京 上海 深圳 新疆  广州 成都  长沙 武汉 .....
        String ragMsg = "今天长沙的天气是晴，20~26度，湘潭的天气的晴，温度22~27度";
        // Token 消耗很多 ，大模型理解需要耗费更多的时间
        String prompt = ragMsg+":今天长沙的天气怎么样？适合哪些活动？";
        float[] embedding = LLMUtils.embedding(prompt, null);
        System.out.println(embedding.length);
        for (float v : embedding) {
            System.out.println(v);
        }
    }
}
